Counter function in Python